In [ ]:
import cv2
from matplotlib import pyplot as plt
import pandas as pd
In [ ]:
fig = plt.figure(figsize=(10, 7))
rows = 2
columns = 2
<Figure size 720x504 with 0 Axes>
In [ ]:
df = pd.read_csv("dataset.csv")
df
Out[ ]:
Path Label Note
0 DSC_2701.JPG 1 NaN
1 DSC_2702.JPG 1 NaN
2 DSC_2703.JPG 2 NaN
3 DSC_2704.JPG 2 NaN
4 DSC_2706.JPG 2 NaN
... ... ... ...
117 IMG_9429.jpg 11 NaN
118 IMG_9430.jpg 11 NaN
119 IMG_9432.jpg 3 NaN
120 IMG_9433.jpg 3 NaN
121 IMG_9435.jpg 5 NaN

122 rows × 3 columns

In [ ]:
print(len(df['Label']))
122
In [ ]:
switch = {
    1: "SuknelÄ—",
    2: "Å varkas",
    3: "Trumpas topas",
    4: "Ilgas topas",
    5: "Sijonas",
    6: "Suknele be rankovių",
    7: "SkarelÄ—",
    8: "Paltas",
    9: "Striuke",
    10: "Marškiniai",
    11: "KelnÄ—s"
}
In [ ]:
# max 122
 
skip = 0
load = 121
columns = 5

fig = plt.figure(figsize=(28, load))

rows = int(load / 4)

for i in range(load):
    index = i + skip
    fig.add_subplot(rows, columns, index + 1)
    plt.imshow(cv2.imread("imgs/" + df['Path'][index]))
    plt.axis('off')
    plt.title(switch.get(int(df['Label'][index])))
In [ ]: